#include "udp_client.hpp"
#include <unistd.h>

using namespace std;

// 使用手册
static void usage(std::string proc)
{
    std::cout << "Usage:\n\t" << proc << " serverip serverport\n"
              << std::endl;
}

//./udp_client server_ip server_port :得知道服务端ip和端口号
int main(int argc, char *argv[])
{
    if (argc != 3)
    {
        usage(argv[0]);
        exit(USAGE_ERR);
    }
    std::string serverip = argv[1];      // ip
    uint16_t serverport = atoi(argv[2]); // 端口号：字符串转16位整形

    // 创建套接字文件描述符
    uint16_t sock = socket(AF_INET, SOCK_DGRAM, 0);
    if (sock < 0)
    {
        std::cerr << "create socket error" << std::endl;
        exit(SOCKET_ERR);
    }

    // socket通信的本质[clientip:clientport, serverip:serverport]原ip、原端口号 + 目的ip、目的端口号
    // 在客户端中是不需要去绑定指定的端口号的，这个工作通常由操作系统来实现
    // 这是因为：客户上的app诸多，每个客户端都要求指定端口号的话，就会造成端口号不足，客户端启动冲突
    // 服务端不一样，要让诸多用户找到同一台服务器就要指定特定的端口号

    // 访问指定的服务器需要知道服务器的ip+端口号：app内部已经实现
    struct sockaddr_in server;
    memset(&server, 0, sizeof(server));

    server.sin_family = AF_INET;                          // 使用ipv4
    server.sin_port = htons(serverport);                  // 主机转网络
    server.sin_addr.s_addr = inet_addr(serverip.c_str()); // 将点分十进制的IP地址字符串转换成一个长整型数

    while (true)
    {
        // 用户输入
        std::string message;
        std::cout << "[kunkun的服务器]# ";
        // std::cin >> message;
        std::getline(std::cin, message); // 从键盘中获取一行字符串，存储到message中

        // 客户端首次次系统调用发送数据的时候，OS会在底层随机选择客户端的端口号+自己的ip
        //  发送数据
        if (sendto(sock, message.c_str(), message.size(), 0, (struct sockaddr *)&server, sizeof(server)) > 0)
        {
            cout << "发送数据成功！" << endl;
        }

        // 接收数据
        char buffer[2048];
        struct sockaddr_in temp; // 用于接收服务端的ip与port
        memset(&temp, 0, sizeof(temp));
        socklen_t len = sizeof(temp);
        int n = recvfrom(sock, buffer, sizeof(buffer) - 1, 0, (struct sockaddr *)&temp, &len);
        // cout << "开始接收数据！" << n << endl;
        sleep(1);
        if (n > 0)
        {
            buffer[n] = '\0';
            // std::cout << "server echo# " << buffer << std::endl;
            std::cout << buffer << std::endl;
        }
    }

    return 0;
}